from torchvision import datasets, transforms
import config
import torch.utils.data


opt = config.DefaultConfig()
transform = transforms.Compose([
    transforms.Resize(opt.img_size),
    transforms.ToTensor()
])

train_loader = torch.utils.data.DataLoader(
    datasets.MNIST("./data", train=True, download=True, transform=transform),
    batch_size=opt.batch_size,
    shuffle=True,
    num_workers=opt.num_work
)






